package NO11_DP;

/**
 * 2025-09-01
 * 力扣 - 198. 打家劫舍
 * <p>
 * 你是一个专业的小偷，计划偷窃沿街的房屋。
 * 每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，
 * 如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。
 * 给定一个代表每个房屋存放金额的非负整数数组，计算你 不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。
 */
public class RobMain {
    public static void main(String[] args) {
        System.out.println(rob(new int[]{1, 2, 3, 1}));
        System.out.println(rob(new int[]{2, 7, 9, 3, 1}));
    }

    /**
     * dp[i] 表示偷窃到第 i 个房屋时，能获得的最高金额
     * 状态转移方程：
     * 对于第 i 个房屋，有两种选择：
     * 不偷第 i 个房屋：那么最高金额等于偷窃到第 i-1 个房屋时的最高金额，即 dp[i] = dp[i-1]
     * 偷第 i 个房屋：那么不能偷第 i-1 个房屋，因此最高金额等于偷窃到第 i-2 个房屋时的最高金额加上当前房屋的金额，即 dp[i] = dp[i-2] + nums[i]
     * dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
     */
    public static int rob(int[] nums) {
        int n = nums.length;
        if (n == 0) {
            return 0;
        }
        if (n == 1) {
            return nums[0];
        }
        if (n == 2) {
            return Math.max(nums[0], nums[1]);
        }

        int[] dp = new int[n];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < n; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        return dp[n - 1];
    }

    // 空间优化，用两个变量来记录前两个状态，将空间复杂度优化到 O(1)
    public static int rob1(int[] nums) {
        int n = nums.length;
        if (n == 0) return 0;
        if (n == 1) return nums[0];

        // 初始化前两个状态
        int prev2 = nums[0];               // dp[0]
        int prev1 = Math.max(nums[0], nums[1]); // dp[1]

        for (int i = 2; i < n; i++) {
            // 计算当前状态：偷当前房屋 vs 不偷当前房屋
            int current = Math.max(prev1, prev2 + nums[i]);

            // 更新状态：向前移动
            prev2 = prev1;
            prev1 = current;
        }

        return prev1;
    }
}
